package com.qf.utils;

import com.qf.entity.UserEmp;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;

import java.io.*;
import java.net.URLEncoder;
import java.util.List;

/**
 * 2019/11/22
 * 郭程鹏
 * gcp1909
 * 面向代码面向君,不负时光不负卿
 */

public class ExcleUtil {
        //1、创建一个excle文件
         static HSSFWorkbook hssfWorkbook;
         //2、创建文件中的sheet表，参数是sheet名称
         static HSSFSheet sheet;
         //创建名，创建表的第一行
        public static void createFirstRow(String sheetName){
            hssfWorkbook=new HSSFWorkbook();
            sheet=hssfWorkbook.createSheet(sheetName);
            //创建第一行的字段名
            HSSFRow row = sheet.createRow(0);
            String[] heads={"账号","姓名","角色","部门"};
            for (int i = 0; i < heads.length; i++) {
                //创建列
                HSSFCell cell = row.createCell(i);
                //每一列的值
                cell.setCellValue(heads[i]);
            }
        }

        //3、循环创建其他行的值 导出数据
    public static void createSheetValue(List<UserEmp>userEmps){

        //向这个表中的其他行赋值
        for (int i = 0; i < userEmps.size(); i++) {
            //创建第二行及以后
            HSSFRow row = sheet.createRow(i + 1);
            //当前行的值
            UserEmp userEmp = userEmps.get(i);
            //向第二行及以后的每一列中赋值

            //第二行第一列
            HSSFCell cell1 = row.createCell(0);
            cell1.setCellValue(userEmp.getLoginNumber());
            //第二行第二列
            HSSFCell cell2 = row.createCell(1);
            cell2.setCellValue(userEmp.getUserName());
            //第二行第三列
            HSSFCell cell3 = row.createCell(2);
            cell3.setCellValue(userEmp.getUserRole().getRoleName());
            //第二行第四列
            HSSFCell cell4 = row.createCell(3);
            cell4.setCellValue(userEmp.getDepartment().getDeptName());
        }
    }

    //excle输出 字节输出
    public static void export(OutputStream outputStream) throws IOException {

        //以网格的格式输出
        sheet.setGridsPrinted(true);
        //创建一个文件
        //以字节流响应给用户
        hssfWorkbook.write(outputStream);
        outputStream.flush();
    }


}
